# pip install zhipuai 请先在终端进行安装

import zhipuai

zhipuai.api_key = "41805a5412196acfdd9d63a905f9039a.Z0TWy2173NoP1ZkI"
response = zhipuai.model_api.sse_invoke(
    model="chatglm_lite",
    prompt= [{"role":"user","content":"请你扩写下面的议论文：\n在宽阔又狭隘的人生道路上，有无数条路，任你选。但一定有一条路，留下了你的足迹。也许，你会像蜜蜂一样勤劳努力，也许，你会像大树一样默默奉献，也许，你会像静物一样碌碌终生。这都是你自己的选择。\n踏哪一条路，都会决定你一生的道路!\n我希望，能把荒草萋萋的小道，开拓成宽阔而平坦的大陆!"}],
    temperature= 0.9,
    top_p= 0.7,
    incremental=True
)

for event in response.events():
    if event.event == "add":
        print(event.data, end="")
    elif event.event == "error" or event.event == "interrupted":
        print(event.data, end="")
    elif event.event == "finish":
        print(event.data)
        print(event.meta, end="")
    else:
        print(event.data, end="")